/*
 *                    GNU LESSER GENERAL PUBLIC LICENSE
 *                        Version 3, 29 June 2007
 *
 *  Copyright (C) 2007 Free Software Foundation, Inc. <http://fsf.org/>
 *  Everyone is permitted to copy and distribute verbatim copies
 *  of this license document, but changing it is not allowed.
 *
 *
 *   This version of the GNU Lesser General Public License incorporates
 * the terms and conditions of version 3 of the GNU General Public
 * License, supplemented by the additional permissions listed below.
 *
 *   0. Additional Definitions.
 *
 *   As used herein, "this License" refers to version 3 of the GNU Lesser
 * General Public License, and the "GNU GPL" refers to version 3 of the GNU
 * General Public License.
 *
 *   "The Library" refers to a covered work governed by this License,
 * other than an Application or a Combined Work as defined below.
 *
 *   An "Application" is any work that makes use of an interface provided
 * by the Library, but which is not otherwise based on the Library.
 * Defining a subclass of a class defined by the Library is deemed a mode
 * of using an interface provided by the Library.
 *
 *   A "Combined Work" is a work produced by combining or linking an
 * Application with the Library.  The particular version of the Library
 * with which the Combined Work was made is also called the "Linked
 * Version".
 *
 *   The "Minimal Corresponding Source" for a Combined Work means the
 * Corresponding Source for the Combined Work, excluding any source code
 * for portions of the Combined Work that, considered in isolation, are
 * based on the Application, and not on the Linked Version.
 *
 *   The "Corresponding Application Code" for a Combined Work means the
 * object code and/or source code for the Application, including any data
 * and utility programs needed for reproducing the Combined Work from the
 * Application, but excluding the System Libraries of the Combined Work.
 *
 *   1. Exception to Section 3 of the GNU GPL.
 *
 *   You may convey a covered work under sections 3 and 4 of this License
 * without being bound by section 3 of the GNU GPL.
 *
 *   2. Conveying Modified Versions.
 *
 *   If you modify a copy of the Library, and, in your modifications, a
 * facility refers to a function or data to be supplied by an Application
 * that uses the facility (other than as an argument passed when the
 * facility is invoked), then you may convey a copy of the modified
 * version:
 *
 *    a) under this License, provided that you make a good faith effort to
 *    ensure that, in the event an Application does not supply the
 *    function or data, the facility still operates, and performs
 *    whatever part of its purpose remains meaningful, or
 *
 *    b) under the GNU GPL, with none of the additional permissions of
 *    this License applicable to that copy.
 *
 *   3. Object Code Incorporating Material from Library Header Files.
 *
 *   The object code form of an Application may incorporate material from
 * a header file that is part of the Library.  You may convey such object
 * code under terms of your choice, provided that, if the incorporated
 * material is not limited to numerical parameters, data structure
 * layouts and accessors, or small macros, inline functions and templates
 * (ten or fewer lines in length), you do both of the following:
 *
 *    a) Give prominent notice with each copy of the object code that the
 *    Library is used in it and that the Library and its use are
 *    covered by this License.
 *
 *    b) Accompany the object code with a copy of the GNU GPL and this license
 *    document.
 *
 *   4. Combined Works.
 *
 *   You may convey a Combined Work under terms of your choice that,
 * taken together, effectively do not restrict modification of the
 * portions of the Library contained in the Combined Work and reverse
 * engineering for debugging such modifications, if you also do each of
 * the following:
 *
 *    a) Give prominent notice with each copy of the Combined Work that
 *    the Library is used in it and that the Library and its use are
 *    covered by this License.
 *
 *    b) Accompany the Combined Work with a copy of the GNU GPL and this license
 *    document.
 *
 *    c) For a Combined Work that displays copyright notices during
 *    execution, include the copyright notice for the Library among
 *    these notices, as well as a reference directing the user to the
 *    copies of the GNU GPL and this license document.
 *
 *    d) Do one of the following:
 *
 *        0) Convey the Minimal Corresponding Source under the terms of this
 *        License, and the Corresponding Application Code in a form
 *        suitable for, and under terms that permit, the user to
 *        recombine or relink the Application with a modified version of
 *        the Linked Version to produce a modified Combined Work, in the
 *        manner specified by section 6 of the GNU GPL for conveying
 *        Corresponding Source.
 *
 *        1) Use a suitable shared library mechanism for linking with the
 *        Library.  A suitable mechanism is one that (a) uses at run time
 *        a copy of the Library already present on the user's computer
 *        system, and (b) will operate properly with a modified version
 *        of the Library that is interface-compatible with the Linked
 *        Version.
 *
 *    e) Provide Installation Information, but only if you would otherwise
 *    be required to provide such information under section 6 of the
 *    GNU GPL, and only to the extent that such information is
 *    necessary to install and execute a modified version of the
 *    Combined Work produced by recombining or relinking the
 *    Application with a modified version of the Linked Version. (If
 *    you use option 4d0, the Installation Information must accompany
 *    the Minimal Corresponding Source and Corresponding Application
 *    Code. If you use option 4d1, you must provide the Installation
 *    Information in the manner specified by section 6 of the GNU GPL
 *    for conveying Corresponding Source.)
 *
 *   5. Combined Libraries.
 *
 *   You may place library facilities that are a work based on the
 * Library side by side in a single library together with other library
 * facilities that are not Applications and are not covered by this
 * License, and convey such a combined library under terms of your
 * choice, if you do both of the following:
 *
 *    a) Accompany the combined library with a copy of the same work based
 *    on the Library, uncombined with any other library facilities,
 *    conveyed under the terms of this License.
 *
 *    b) Give prominent notice with the combined library that part of it
 *    is a work based on the Library, and explaining where to find the
 *    accompanying uncombined form of the same work.
 *
 *   6. Revised Versions of the GNU Lesser General Public License.
 *
 *   The Free Software Foundation may publish revised and/or new versions
 * of the GNU Lesser General Public License from time to time. Such new
 * versions will be similar in spirit to the present version, but may
 * differ in detail to address new problems or concerns.
 *
 *   Each version is given a distinguishing version number. If the
 * Library as you received it specifies that a certain numbered version
 * of the GNU Lesser General Public License "or any later version"
 * applies to it, you have the option of following the terms and
 * conditions either of that published version or of any later version
 * published by the Free Software Foundation. If the Library as you
 * received it does not specify a version number of the GNU Lesser
 * General Public License, you may choose any version of the GNU Lesser
 * General Public License ever published by the Free Software Foundation.
 *
 *   If the Library as you received it specifies that a proxy can decide
 * whether future versions of the GNU Lesser General Public License shall
 * apply, that proxy's public statement of acceptance of any version is
 * permanent authorization for you to choose that version for the
 * Library.
 *
 */

package vip.isass.core.support;

import cn.hutool.core.map.MapUtil;
import org.springframework.beans.BeansException;
import org.springframework.beans.factory.support.BeanDefinitionBuilder;
import org.springframework.beans.factory.support.DefaultListableBeanFactory;
import org.springframework.context.ApplicationContext;
import org.springframework.context.ApplicationContextAware;
import org.springframework.context.ConfigurableApplicationContext;
import org.springframework.core.ParameterizedTypeReference;
import org.springframework.core.ResolvableType;
import org.springframework.stereotype.Component;

import java.util.Collection;
import java.util.Collections;
import java.util.Map;

/**
 * @author Rain
 */
@Component
public class SpringContextUtil implements ApplicationContextAware {

    private static ApplicationContext applicationContext;

    @Override
    public void setApplicationContext(ApplicationContext applicationContext) throws BeansException {
        SpringContextUtil.applicationContext = applicationContext;
    }

    public static ApplicationContext getApplicationContext() {
        return applicationContext;
    }

    /**
     * @return 是否已经初始化 spring 环境
     */
    public static boolean isInitialized() {
        return applicationContext != null;
    }

    /**
     * 创建一个java对象，并添加到spring，让spring管理
     *
     * @param <T>       bean class
     * @param beanClass class to new instance
     * @return spring bean
     */
    public static <T> T addBeanToSpringContext(Class<T> beanClass) {
        //获取BeanFactory
        DefaultListableBeanFactory defaultListableBeanFactory = (DefaultListableBeanFactory) applicationContext.getAutowireCapableBeanFactory();

        //创建bean信息
        String beanName = getBeanNameByBeanType(beanClass);
        BeanDefinitionBuilder beanDefinitionBuilder = BeanDefinitionBuilder.genericBeanDefinition(beanClass);

        //动态注册bean
        defaultListableBeanFactory.registerBeanDefinition(beanName, beanDefinitionBuilder.getBeanDefinition());

        @SuppressWarnings("unchecked")
        T bean = (T) applicationContext.getBean(beanName);
        return bean;
    }

    /**
     * 根据name获取bean
     *
     * @param name bean name
     * @return bean
     */
    public static Object getBean(String name) throws BeansException {
        return applicationContext.getBean(name);
    }

    /**
     * 根据name获取bean
     *
     * @param <T>          require type
     * @param name         name
     * @param requiredType required type
     * @return bean
     */
    public static <T> T getBean(String name, Class<T> requiredType) {
        return applicationContext.getBean(name, requiredType);
    }

    /**
     * 根据bean类型获取bean
     *
     * @param <T>          require type
     * @param requiredType required type
     * @return bean
     */
    public static <T> T getBean(Class<T> requiredType) {
        return applicationContext.getBean(requiredType);
    }

    public static <T> Collection<T> getBeans(Class<T> requiredType) {
        Map<String, T> beans = applicationContext.getBeansOfType(requiredType);
        if (MapUtil.isEmpty(beans)) {
            return Collections.emptyList();
        }

        return beans.values();
    }

    public static <T> Collection<T> getBeans(Class<T> requiredType, ParameterizedTypeReference<T> parameterizedTypeReference) {
        Map<String, T> beans = applicationContext.getBeansOfType(requiredType);
        if (MapUtil.isEmpty(beans)) {
            return Collections.emptyList();
        }

        return beans.values();
    }

    public static <T, P> T getBean(Class<T> requiredType, Class<P> type) {
        Map<String, T> beans = applicationContext.getBeansOfType(requiredType);
        if (MapUtil.isEmpty(beans)) {
            return null;
        }

        for (T bean : beans.values()) {
            ResolvableType resolvableType = ResolvableType.forClass(bean.getClass());
            System.out.println(resolvableType);
            Class<?> resolve = resolvableType.getSuperType().getGeneric(0).resolve();
            System.out.println(resolve);
            if (type.equals(resolve)) {
                return bean;
            }
        }
        return null;
    }

    public static <T, P> T getBeanOfSupport(Class<T> requiredType, Class<P> supportType) {
        Map<String, T> beans = applicationContext.getBeansOfType(requiredType);
        if (MapUtil.isEmpty(beans)) {
            return null;
        }

        for (T bean : beans.values()) {
            if (!(bean instanceof Support)) {
                continue;
            }
            if (((Support) bean).support(supportType)) {
                return bean;
            }
        }
        return null;
    }

    /**
     * 根据bean类型，构造器参数获取bean
     *
     * @param <T>          require type
     * @param requiredType required type
     * @param objects      objects
     * @return bean
     */
    public static <T> T getBean(Class<T> requiredType, Object... objects) {
        return applicationContext.getBean(requiredType, objects);
    }

    /**
     * 根据类名获取bean
     *
     * @param <T>       require type
     * @param beanClass bean class
     * @return bean
     */
    public static <T> T getBeanByNameOfBeanType(Class<T> beanClass) {
        String beanName = getBeanNameByBeanType(beanClass);
        @SuppressWarnings("unchecked")
        T bean = (T) applicationContext.getBean(beanName);
        return bean;
    }

    /**
     * 根据非限定类名获取bean的名称，即将类名的首字母小写
     *
     * @param beanClass bean class
     * @return bean name
     */
    public static String getBeanNameByBeanType(Class<?> beanClass) {
        String beanName = beanClass.getSimpleName();
        return beanName.substring(0, 1).toLowerCase() + beanName.substring(1);
    }

    /**
     * 移除 bean
     *
     * @param beanName bean name
     */
    public static void unRegistryBean(String beanName) {
        DefaultListableBeanFactory beanFactory = getBeanFactory();
        if (beanFactory.containsBeanDefinition(beanName)) {
            beanFactory.destroySingleton(beanName);
            beanFactory.removeBeanDefinition(beanName);
        }
    }

    /**
     * 移除 bean
     *
     * @param beanName bean name
     */
    public static void unRegistryBean(Class<?> beanClass) {
        String[] beanNames = applicationContext.getBeanNamesForType(beanClass);
        for (String beanName : beanNames) {
            unRegistryBean(beanName);
        }
    }

    /**
     * 获取 SpringBean 注册器
     *
     * @return BeanDefinitionRegistry
     */
    private static DefaultListableBeanFactory getBeanFactory() {
        ConfigurableApplicationContext configurableApplicationContext = (ConfigurableApplicationContext) applicationContext;
        return (DefaultListableBeanFactory) configurableApplicationContext.getBeanFactory();
    }

    /**
     * 获取实现某接口的所有实例 bean 名称
     *
     * @param type 接口
     * @return beanName 数组
     */
    public static String[] getBeanNamesForType(Class<?> type) {
        return applicationContext.getBeanNamesForType(type);
    }

}
